package com.bright.ghj.common.mybatis;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.toolkit.ArrayUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.converts.SqliteTypeConvert;
import com.baomidou.mybatisplus.generator.config.po.TableField;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.IColumnType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;
import com.bright.ghj.common.mybatis.pojo.BaseEntity;

import java.util.ArrayList;

/**
 * @author xiaozq
 * @date 2023/6/1 11:23
 * mybatis-plus code generator.
 **/
public class MybatisPlusGenerator {

    //生成文件所在项目路径
//    private static String baseProjectPath = "E:\\ideaProject\\hxj\\bj-new\\qchz-receive\\src\\main\\java";
//
//    //基本包名
//    private static String basePackage = "com.bright.hxj.qchz";
//
//    //作者(一般公司名简称)
//    private static String authorName = "hxj";
//
//    //要生成的表名，为空时生成所有表
//    private static String[] tables = {"Duijie_Baowen_log"};
//
//    //table前缀:生成代码的时候可以忽略前缀，生成实体类：
//    /**例如: private static String prefix = "t_";
//     则t_goods_info的表，生成实体类则是GoodsInfo而不是TGoodsInfo
//     */
//    private static String prefix = "";
//
//    //数据库配置四要素
//    private static String driverName = "com.microsoft.sqlserver.jdbc.SQLServerDriver";
//    private static String url = "jdbc:sqlserver://219.135.179.253:62224;DataBaseName=zcjy_szqc2022_mz;encrypt=false";
//    private static String username = "sa";
//    private static String password = "gzbright38259201@2w3e";
////    private static String url = "jdbc:sqlserver://localhost:1433;DataBaseName=zcjy_th_duijie_gzs_jieshou;encrypt=false";
////    private static String username = "sa";
////    private static String password = "gzbright38259201@";
////    private static String url = "jdbc:sqlserver://219.135.179.253:62224;DataBaseName=analysis_szqc_nb_new;encrypt=false";
////    private static String username = "sa";
////    private static String password = "gzbright38259201@2w3e";

    public static void start(String baseProjectPath, String authorName, String url, String driverName, String username, String password,
                             String basePackage, String[] tables, String prefix) {
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();

        // 1、全局配置
        GlobalConfig gc = new GlobalConfig();

        //可以将生成得代码单独放在文件夹里，检查完没问题在手动将包复制到项目中
        gc.setOutputDir(baseProjectPath);
        gc.setAuthor(authorName);// 生成实体类的文档注解中作者名
        gc.setOpen(true);// 生成代码后自动打开生成目录文件
        gc.setFileOverride(false); // 是否覆盖原来生成的
        gc.setIdType(IdType.AUTO);// 指定生成的主键类型：自增长
        gc.setDateType(DateType.ONLY_DATE);
        gc.setActiveRecord(false);
        mpg.setGlobalConfig(gc);

        // 2、数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(url);
        dsc.setDriverName(driverName);
        dsc.setUsername(username);
        dsc.setPassword(password);
        dsc.setDbType(DbType.SQL_SERVER); // mysql数据库
        gc.setServiceName("%sService"); //去掉Service接口的首字母'I'
        dsc.setTypeConvert(new SqliteTypeConvert() {
            @Override
            public IColumnType processTypeConvert(GlobalConfig globalConfig, TableField tableField) {
                return super.processTypeConvert(globalConfig, tableField);
            }
        });
        mpg.setDataSource(dsc);

        // 3、包配置
        PackageConfig pc = new PackageConfig();
        pc.setModuleName(null)// 生成的controller路径映射为一个"/",覆盖默认的"//"
                .setParent(basePackage)// 生成controller，service，xml并设置路径。
                .setEntity("pojo.po")
                .setMapper("mapper")
                .setService("service")
                .setServiceImpl("service.impl")
                .setController("controller2");
        mpg.setPackageInfo(pc);

        // 5、策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setCapitalMode(true);// 开启全局大写命名
        strategy.setNaming(NamingStrategy.underline_to_camel);// 实体类名驼峰
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);// 属性名驼峰
        strategy.setInclude(tables);// 设置要映射的表名注解@Table
        strategy.setSuperEntityClass(BaseEntity.class); // 预先定义一个BaseEntity类实现Serializable接口
        strategy.setEntityLombokModel(true); // 使用 lombok
        strategy.setRestControllerStyle(true);// 开启 restful风格
        strategy.setEntityTableFieldAnnotationEnable(false);// 不生成实体类字段swagger类型注解,而是采用文档类型注解：/**xxx*/

        if (prefix != null) strategy.setTablePrefix(prefix);  // 表前缀
        if (ArrayUtils.isNotEmpty(tables)) strategy.setInclude(tables);  // 表名

        // 6、自动填充配置：非必须
        TableFill createTime = new TableFill("createDate", FieldFill.INSERT);
        TableFill updateTime = new TableFill("writeDate", FieldFill.INSERT_UPDATE);
        ArrayList<TableFill> tableFills = new ArrayList<>();
        tableFills.add(createTime);
        tableFills.add(updateTime);
        strategy.setTableFillList(tableFills);
        strategy.setLogicDeleteFieldName("deleted");// 指定逻辑删除字段

        mpg.setStrategy(strategy);

        mpg.setTemplateEngine(new VelocityTemplateEngine());

        // 执行生成器
        mpg.execute();
    }

}


